import { crawlNovel } from './src/crawler.js';
import { saveNovelToFile } from './src/fileManager.js';
import path from 'path';
import { fileURLToPath } from 'url';

// 获取当前文件的目录
const __dirname = path.dirname(fileURLToPath(import.meta.url));

// 小说配置
const config = {
    url: 'https://m.3yt.org/ml/152842/',
    outputFile: path.join(__dirname, 'novels', '3yt-test.txt'),
    startChapter: 1,  // 从第1章开始
    maxChapters: 1055,   // 最多爬取3章
    delay: 100     // 每章之间延迟2秒
};

console.log('3yt.org...');
console.log(`目录页URL: ${config.url}`);
console.log(`输出文件: ${config.outputFile}`);
console.log(`起始章节: ${config.startChapter}`);
console.log(`最大章节数: ${config.maxChapters}`);
console.log(`爬取间隔: ${config.delay}ms`);
console.log('----------------------------');

async function main() {
    try {
        // 爬取小说
        const novel = await crawlNovel(
            config.url,
            config.startChapter,
            config.maxChapters,
            config.delay
        );

        // 保存为文件
        await saveNovelToFile(novel, config.outputFile);

        console.log('爬取完成！');
    } catch (error) {
        console.error('爬取过程中出错:', error);
    }
}

main();